快速精准的人头检测，代码已开源
点击我爱计算机视觉标星，更快获取CVML新技术
昨天arXiv一篇新上论文《FCHD: A fast and accurate head detector 》，来自江森自控（Johnson Controls Inc.）的软件工程师Aditya Vora分享了一种快速精准的人头检测（head detector）算法并开源了代码。
先来看看作者发布的视频效果：
看起来还是不错的！
人头检测在视频监控中非常重要，而公交车、商场或者大型场馆的拥挤人群计数则是其重要应用场景。
算法思想
作者称拥挤人群计数目前主要有两种实现路径：
1.使用回归的算法思路，直接根据图像回归出拥挤人群密度热图，它的缺点是只能得到场景整体的一个拥挤指数，不能获知人群个体的具体位置，而且这种方法对图像分辨率很敏感。（52CV君曾经分享过：尺度不变网络提升人群计数性能（附Github地址））
2.使用目标检测的方法，比如直接使用Faster RCNN检测人，检测后数目标为“人”的个数。这种方法的缺点是在人物相互遮挡的情况下往往性能较差，而人群越拥挤相互遮挡的可能性越大，导致算法使用受限。
该文作者希望设计更有针对性的精准的人头检测，实现更加精准的人群计数。
作者创新的两点，轻量级人头检测网络和anchors尺度的选择。
网络架构：
从上图可知，网络的前半部分是VGG16特征提取网络，经过一次卷积，然后分开分别使用1*1卷积进入Regression head（用于回归人头的位置信息）和Classification head（用于分类是否为人头）。
在检测网络中，anchors是很重要的概念，是一组预定义的包围框，在对象检测系统中预测尺度和位移。作者认为普通目标识别中anchors的尺度定义过大，应该根据“有效感受野”（effective receptive field）的思想，减小尺度（直观上“人头”目标比“人体”目标也要小）。
实验结果
作者在BRAINWASH数据库上做了实验，打败了三个基线算法，并取得了与最好结果算法“可比较“的性能。
提出的算法比较快，使用Quadro M1000M显卡（仅有512 CUDA核心）速度达到5fps，使用Jetson TX2（仅有256 CUDA核心）速度1.6fps。相比以往算法更加适合于嵌入式边缘计算平台。
论文地址：
https://arxiv.org/abs/1809.08766v1
代码地址：
https://github.com/aditya-vora/FCHD-Fully-Convolutional-Head-Detector
论文、代码、数据下载：
在“我爱计算机视觉”公众号对话界面回复“FCHD”，即可收到百度云下载地址。
长按关注我爱计算机视觉
【点赞与转发】就是一种鼓励！